torch.erf 和torch.bmm 和sub_算子报错 #463 ... 你好，这个关键性算子torch.erf还是不可以，其他两个算子可以通过别的方式代替，但这个算子还是之前 ... ... <看更多>

You can use torch.bmm (https://pytorch.org/docs/stable/torch.html#torch.bmm). Just do torch.bmm(I,W). To verify the results : ... <看更多>

torch.bmm #111. Hi, I can't find bmm anywhere, is it not supported? Any plans to add it? alexbw. alexbw COLLABORATOR. Created 5 years ago. ... <看更多>

... res[1,:]=a[1,:] x b[1,;] 其中x表示矩阵乘法res = torch.matmul(a, b) # 维度res=[2,3,2] print(res) # res2的值等于res res2 = torch.bmm(a, ... ... <看更多>

... <看更多>